from index import create_index, search
from utils import setup_logger


def main():
    # 初始化日志记录器
    logger = setup_logger()

    # 询问用户输入文件索引的目录路径
    directory = input("请输入要索引的文件目录路径: ")

    # 从用户指定的目录创建索引
    logger.info(f"正在从以下目录创建索引: {directory}")
    index, files = create_index(directory)
    logger.info("索引创建完成。")

    # 搜索循环
    try:
        while True:
            query = input("请输入搜索查询: ").strip()
            if query:
                results = search(query, index, files)
                if results:
                    print("找到以下文件:")
                    for result in results:
                        print(result)
                else:
                    print("未找到相关文件。")
            else:
                print("请输入有效的搜索查询。")
    except KeyboardInterrupt:
        print("\n程序已中断。")


if __name__ == "__main__":
    main()
